"""
    目标：
        截屏
    方法：
        driver.get_screenshot_as_file()
    需求：
        1. 输入用户名
        2. 截图当前目录下 admin.png
"""
import time

# 导包
from selenium import webdriver
from time import sleep

from selenium.webdriver.common.by import By

# 获取浏览器驱动对象
driver = webdriver.Chrome()

# 打开url
url = r"D:\python\python-selenium-learn\a05_HTML特殊元素处理\learn-files\注册A.html"
driver.get(url)

# 最大化浏览器
driver.maximize_window()

# 输入admin
driver.find_element(By.CSS_SELECTOR, "#userA").send_keys("admin")

# 调用截图方法
# driver.get_screenshot_as_file("./learn-files/admin.png")
# 存放指定目录
# driver.get_screenshot_as_file("../image/admin.png")
# 动态获取文件名称，使用时间戳
driver.get_screenshot_as_file(f"./learn-files/{time.strftime('%Y_%m_%d %H_%M_%S')}.png")

# 暂停2秒
sleep(2)
# 关闭驱动对象
driver.quit()
